package org.proddevplm.dao.data.diagram;


import java.util.HashMap;
import java.util.Map;

public class EntityStructure {


    Map<String, DiagramEntity> entityMap;

    public Map<String, DiagramEntity> getEntityMap() {
        if (entityMap == null) {
            entityMap = new HashMap<String, DiagramEntity>();
        }
        return entityMap;
    }

    public DiagramEntity getEntity(String key) {
        return entityMap.get(key);
    }

    public void addEntity(DiagramEntity entity) {
        getEntityMap().put(entity.getIdentifier(), entity);
    }
}
